import torch


# Compare this snippet from PyTorchs/4.张量运算方法/1.张量的拼接操作.py:
# import torch
#
data1 = torch.randint(0, 10, [1, 3, 4])
data2 = torch.randint(0, 10, [1, 3, 4])

# 按照0轴拼接
print(torch.cat([data1, data2], dim=0))
# 按照1轴拼接
print(torch.cat([data1, data2], dim=1))
# 按照2轴拼接
new_data = torch.cat([data1, data2], dim=2)

print(new_data.shape)

# todo 总结：
#  torch.cat()函数用于拼接张量，dim参数用于指定拼接的维度，
#  dim=0表示按行拼接，dim=1表示按列拼接，dim=2表示按块拼接。